home *** CD-ROM | disk | FTP | other *** search
/ Games of Daze / Infomagic - Games of Daze (Summer 1995) (Disc 1 of 2).iso / x2ftp / msdos / memory / pmdkt110 / heap.dif < prev    next >
Text File  |  1994-11-14  |  5KB  |  234 lines
  1. *** c:\bp\rtl\sys\heap.asm    Wed Oct 28 07:00:00 1992
  2. --- heap.asm    Sat Oct 22 16:58:30 1994
  3. ***************
  4. *** 20,27 ****
  5.       EXTRN    FreeList:DWORD,HeapError:DWORD
  6.   
  7. ! ; Local variables
  8.   
  9.   AllocSize    DW    ?
  10.   
  11.   DATA    ENDS
  12.   
  13. --- 20,36 ----
  14.       EXTRN    FreeList:DWORD,HeapError:DWORD
  15.   
  16. ! ;;MemCheck externals
  17.   
  18. +         EXTRN   CheckMEM:BYTE,StoreAlloc:DWORD,FreeAlloc:DWORD
  19. + 
  20. + 
  21. + ;; Local variables
  22. + 
  23.   AllocSize    DW    ?
  24.   
  25. + ;; Memchecker variable
  26. + 
  27. + MemSize         DW      ?
  28. + 
  29.   DATA    ENDS
  30.   
  31. ***************
  32. *** 39,42 ****
  33. --- 48,52 ----
  34.       PUBLIC    GetFreMem,GetFreMax,NewMemory,DisMemory
  35.   
  36. + 
  37.   ; New and GetMem standard procedures
  38.   
  39. ***************
  40. *** 161,167 ****
  41.   NewMemory:
  42.   
  43.       OR    AX,AX
  44. !     JE    @@9
  45. !     MOV    AllocSize,AX
  46.   @@1:    CALL    PointerSize
  47.       MOV    DI,OFFSET DS:FreeList
  48. --- 171,200 ----
  49.   NewMemory:
  50.   
  51. + ;; memchecker code
  52. +         mov     [MemSize],ax
  53. + 
  54.       OR    AX,AX
  55. ! 
  56. ! ;; this jump was too far, so changed by memchecker
  57. ! ;;        JE      @@9
  58. !         jne     @@m0
  59. !         jmp     @@9
  60. ! 
  61. ! ;; memchecker code
  62. ! ;; add 8 or 16 bytes to allocated memory for overflow checking
  63. ! 
  64. ! @@m0:   cmp     CheckMem,0             ; is MemCheck code enabled?
  65. !         je      @@m1
  66. !         cmp     ax,65512               ; 65536-8-16
  67. !         ja      @@CheckUpperArea
  68. !         add     ax,16
  69. !         jmp     short @@m1
  70. ! @@CheckUpperArea:
  71. !         cmp     ax,65520               ; 65536-8-8
  72. !         ja      @@m1
  73. !         add     ax,8
  74. ! 
  75. ! @@m1:
  76. !         MOV    AllocSize,AX
  77.   @@1:    CALL    PointerSize
  78.       MOV    DI,OFFSET DS:FreeList
  79. ***************
  80. *** 173,177 ****
  81.       MOV    SI,ES
  82.       CMP    SI,HeapPtr.w2
  83. !     JE    @@6
  84.       CMP    DX,ES:[DI].mSize.w2
  85.       JA    @@2
  86. --- 206,216 ----
  87.       MOV    SI,ES
  88.       CMP    SI,HeapPtr.w2
  89. ! 
  90. ! ;; memchecker changed jump, else out of range
  91. ! ;;        JE      @@6
  92. !         jne     @@m3
  93. !         jmp     short @@6
  94. ! @@m3:
  95. ! 
  96.       CMP    DX,ES:[DI].mSize.w2
  97.       JA    @@2
  98. ***************
  99. *** 213,216 ****
  100. --- 252,261 ----
  101.       POP    DX
  102.       CLC
  103. + 
  104. + ;; memchecker code
  105. +         jc      @@NoCheck1        ; if error don't check
  106. +         call    MemCheckAddPtr
  107. + @@NoCheck1:
  108. + 
  109.   @@5:    RET
  110.   @@9:    XOR    AX,AX
  111. ***************
  112. *** 240,243 ****
  113. --- 285,294 ----
  114.       POP    AX
  115.       POP    DX
  116. + 
  117. + ;; memchecker code
  118. +         jc      @@NoCheck2        ; if error don't check
  119. +         call    MemCheckAddPtr
  120. + @@NoCheck2:
  121. + 
  122.       CLC
  123.       RET
  124. ***************
  125. *** 245,253 ****
  126.       CALL    HeapError
  127.       CMP    AL,1
  128. !     JB    @@5
  129. !     JE    @@9
  130.       MOV    AX,AllocSize
  131.       JMP    @@1
  132.   
  133.   ; Dispose memory
  134.   ; In    AX    = Size
  135. --- 296,353 ----
  136.       CALL    HeapError
  137.       CMP    AL,1
  138. !         JB      @@5
  139. !         JE      @@9
  140.       MOV    AX,AllocSize
  141.       JMP    @@1
  142.   
  143. + ;;memchecker code
  144. + MemCheckAddPtr:
  145. +         cmp     CheckMem,0        ; if MemCheck disabled
  146. +         je      @@NoCheck         ; don't check
  147. +         mov     cx,MemSize
  148. +         jcxz    @@NoCheck         ; if size = 0, don't check
  149. +         cmp     dx,0              ; check if returned pointer is nil
  150. +         jne     @@no_error
  151. +         cmp     ax,0
  152. +         je      @@NoCheck         ; don't check
  153. + 
  154. + @@no_error:
  155. +         cmp     cx,65520          ; is there room for upper memory check?
  156. +         ja      @@StorePointer2
  157. +         mov     si,ax             ; save ax
  158. +         mov     es,dx
  159. +         mov     di,ax
  160. +         add     di,cx
  161. +         cmp     cx,65512
  162. +         ja      @@m22
  163. +         add     di,8              ; correct if we have a lower memory also
  164. + @@m22:  mov     ax,0cccch         ; fill last 8 bytes with 0CCh
  165. +         mov     cx,4
  166. +         cld
  167. +         rep     stosw
  168. +         mov     ax,si             ; restore ax
  169. +         mov     cx,MemSize
  170. + 
  171. +         cmp     cx,65512          ; is there room for lower memory check?
  172. +         ja      @@StorePointer2
  173. +         mov     es,dx
  174. +         mov     di,ax
  175. +         mov     ax,0cccch         ; fill first 8 bytes with 0CCh
  176. +         mov     cx,4
  177. +         cld
  178. +         rep stosw
  179. +         mov     ax,di             ; first 8 bytes are skipped
  180. + 
  181. + @@StorePointer2:
  182. +         push    dx                ; push ptr
  183. +         push    ax
  184. +         push    [MemSize]         ; else push size
  185. +         call    dword ptr [StoreAlloc]
  186. + 
  187. + @@NoCheck:
  188. +         ret
  189. + 
  190. + 
  191. + 
  192.   ; Dispose memory
  193.   ; In    AX    = Size
  194. ***************
  195. *** 257,260 ****
  196. --- 357,385 ----
  197.   DisMemory:
  198.   
  199. + ;; memchecker code
  200. +         cmp     CheckMem,0
  201. +         je      @@DisMemoryStart
  202. +         or      ax,ax                  ; if Size = 0
  203. +         jne     @@m1
  204. +         jmp     short @@4              ; then no check
  205. + @@m1:
  206. +         push    bx                     ; push ptr
  207. +         push    cx
  208. +         push    ax                     ; push size
  209. +         call    dword ptr [FreeAlloc]
  210. + 
  211. + ;; add the extra overflow memory area size
  212. +         cmp     ax,65512               ; 65536-8-16
  213. +         ja      @@UpperArea
  214. +         add     ax,16
  215. +         sub     cx,8
  216. +         jmp     short @@DisMemoryStart
  217. + @@UpperArea:
  218. +         cmp     ax,65520               ; 65536-8-8
  219. +         ja      @@DisMemoryStart
  220. +         add     ax,8
  221. + 
  222. + @@DisMemoryStart:
  223. + 
  224.       OR    AX,AX
  225.       JE    @@6
  226. ***************
  227. *** 367,370 ****
  228. --- 492,496 ----
  229.       AND    AX,8
  230.       RET
  231. + 
  232.   
  233.   CODE    ENDS
  234.